package com.syl.starter.itools.common.utils;

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 最简单的内存缓存
 *
 * @author syl
 * @create 2019-01-04 16:12
 **/
public class SimpleCacheUtils {
    private static Map<String, CacheData> CACHE_DATA = new ConcurrentHashMap<>();

    // 随便的测试方法
//    public static void main(String [] args)throws Exception{
//        String key = "test";
//        String value = SimpleCacheUtils.getData(key, () -> "testValue", 2000);
//
//        System.out.println("value:"+value);
//
//        Thread.sleep(3 * 1000);
//
//        String str = SimpleCacheUtils.getData(key);
//        System.out.println("value1:" + str);
//
//        String value2 = SimpleCacheUtils.getData(key, () -> "什么", 3000);
//        System.out.println("value2:" + value2);
//        System.out.println("value3:" + SimpleCacheUtils.popup(key));
//
//    }

    /**
     * 数据是否存在
     * @param key
     * @return
     */
    public static boolean isExist(String key){
        return getData(key) != null;
    }

    /**
     * 最简单的获取数据
     * @param key
     * @param <T>
     * @return
     */
    public static <T> T getData(String key){
        return popup(key, false);
    }

    /**
     * 获取数据 如果不存在则添加
     * @param key
     * @param load
     * @param expire 过期时间 单位毫秒 0为永久有效
     * @param <T>
     * @return
     */
    public static <T> T getData(String key, Load<T> load, int expire){
        T data = getData(key);
        if(data == null && load != null){
            data = load.load();
            if(data != null){
                setData(key, data, expire);
            }
        }
        return data;
    }

    /**
     * 弹出数据
     * @param key
     * @param <T>
     * @return
     */
    public static <T> T popup(String key){
        return popup(key, true);
    }

    /**
     * 弹出数据
     * @param key
     * @param delete 是否立即删除
     *
     * @return 获取不到时或超时返回空
     */
    public static <T> T popup(String key, boolean delete){
        CacheData<T> data = CACHE_DATA.get(key);
        if(data != null && (data.getExpire() <= 0 || data.getSaveTime() >= new Date().getTime())){
            return data.getData();
        }
        if(delete) delete(key);
        return null;
    }

    /**
     * 设置数据 永久有效
     * @param key
     * @param data
     * @param <T>
     */
    public static <T> void setData(String key, T data){
        setData(key, data, 0);
    }

    /**
     * 设置数据
     * @param key
     * @param data
     * @param expire 过期时间 单位毫秒 0为永久有效
     * @param <T>
     */
    public static <T> void setData(String key, T data, int expire){
        CACHE_DATA.put(key, new CacheData(data, expire));
    }

    /**
     * 删除数据
     * @param key
     */
    public static void delete(String key){
        CACHE_DATA.remove(key);
    }

    /**
     * 清除所有数据 慎用！！
     */
    public static void clearAll(){
        CACHE_DATA.clear();
    }

    public interface Load<T>{
        T load();
    }

    private static class CacheData<T>{
        /**
         * @param t
         * @param expire 毫秒
         */
        CacheData(T t, int expire){
            this.data = t;
            this.expire = expire <= 0 ? 0 : expire;
            this.saveTime = new Date().getTime() + this.expire;
        }
        private T data;
        private long saveTime; // 存活时间
        private long expire;   // 过期时间 小于等于0标识永久存活
        public T getData() {
            return data;
        }
        public long getExpire() {
            return expire;
        }
        public long getSaveTime() {
            return saveTime;
        }
    }

}
